import os
from dotenv import load_dotenv

# 自动加载项目根目录的 .env 文件
load_dotenv()

def get_config():
    return {
        "env": os.getenv("AIPRE_ENV", "prod"),  # 当前运行环境
        "db": {
            "host": os.getenv("DB_HOST"),  # 数据库主机地址
            "port": int(os.getenv("DB_PORT", "3306")),  # 数据库端口号
            "user": os.getenv("DB_USER"),  # 数据库用户名
            "password": os.getenv("DB_PASSWORD"),  # 数据库密码
            "database": os.getenv("DB_NAME")  # 数据库名称
        },
        "model": {
            "output_path": os.getenv("MODEL_OUTPUT_PATH", "models"),  # 模型文件保存路径
            "n_estimators": 100,  # 默认树的数量
            "learning_rate": 0.1  # 默认学习率
        },
        "paths": {
            "feature_path": os.getenv("FEATURE_PATH", "features"),  # 特征文件保存路径
            "prediction_path": os.getenv("PREDICTION_PATH", "predictions")  # 预测结果保存路径
        }
    }